Fractional State Space Transition for Long Sequence Modeling
分数阶状态空间转移用于长序列建模
🧭 长序列建模 · 分数阶状态空间 ★ 深度解读
Huawei Noah's Ark Lab, Montreal Research Center, Canada
用分数阶动力学替换指数遗忘,通过有限指数模态近似实现高效选择性SSM长序列建模
📌 工作参考价值(5 点)
- 用户长期行为序列建模可直接借鉴:电商/广告场景通常有数月甚至跨年的点击、浏览、购买序列,传统 SSM/RNN 的指数衰减会快速抹掉早期兴趣。FRAC 的分数阶动力学生成幂律记忆,能在更长时间范围保留历史影响,适合作为用户长期兴趣记忆模块。
- 多时间尺度记忆银行的具体实现:将状态转移分解为 log-spaced 指数衰减模态,用可学习 α_t 控制衰减强度,并用 token-dependent λ_t 做选择性调制。这种设计与用户行为中不同时间尺度模式(如近期强兴趣、历史稳定偏好)天然对齐,可直接嵌入现有 user encoder。
- 工程实现与训练效率:FRAC 保持一阶仿射递归,可用 chunked parallel scan 并行训练/prefill,自回归解码有界状态,且有 custom Triton kernel。对需要低延迟、长序列增量更新的推荐/广告检索系统来说,架构可迁移。
- 读写对齐技巧:read/write 路径共享 softmax(-α_t log τ_m + g(u)_m) 的对数时间尺度寻址,能提升信息写入和读取的一致性。在用户行为表示中可类比为让不同时间尺度的兴趣槽位在写入和读取时使用同一套寻址逻辑,增强记忆可检索性。
- 若业务序列普遍较短或已由 Transformer 覆盖,收益可能有限;但对长上下文行为建模和实时增量计算场景值得尝试。